A JoLT for the KV Cache: Near-Lossless KV Cache Compression via Joint Tucker and JL-Residual Allocation for LLMs
LLM KV缓存近无损压缩:联合Tucker分解与JL残差分配
🧭 LLM推理 · KV缓存压缩
将KV缓存视为三阶张量,用部分Tucker分解+JL旋转低比特残差联合优化,实现近无损2-3倍压缩
📌 工作参考价值(4 点)
- 对于部署LLM做推荐解释、对话Agent的场景,可直接应用JoLT将KV缓存压缩2-3倍,节省显存、提升批处理大小与吞吐,且保持输出质量几乎不变。
- Tucker分解+JL残差的联合优化思路(拉格朗日对偶分配秩与比特预算)可迁移到电商搜索推荐中其它高维张量压缩任务,如用户-商品-时间嵌入压缩。
- FlashJoLT的随机SVD加速策略可作为在线压缩的工程参考,适用于需要动态调整压缩率的推荐模型部署。
- 该方法在分组查询注意力和多头注意力结构上均有效,证明其对主流LLM架构通用,建议在实际Agent/生成式推荐系统中优先尝试。